python initialize array of size n